/*!
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
import { Box, useDisclosure } from "@chakra-ui/react";
import { useHotkeys } from "react-hotkeys-hook";
import { useTranslation } from "react-i18next";
import { CgRedo } from "react-icons/cg";

import type { LightGridTaskInstanceSummary, TaskInstanceResponse } from "openapi/requests/types.gen";
import { ClearGroupTaskInstanceDialog } from "src/components/Clear/TaskInstance/ClearGroupTaskInstanceDialog";
import { Tooltip } from "src/components/ui";
import ActionButton from "src/components/ui/ActionButton";

import ClearTaskInstanceDialog from "./ClearTaskInstanceDialog";

type Props = {
  readonly groupTaskInstance?: LightGridTaskInstanceSummary;
  readonly isHotkeyEnabled?: boolean;
  // Optional: allow parent to handle opening a stable, page-level dialog
  readonly onOpen?: (ti: LightGridTaskInstanceSummary | TaskInstanceResponse) => void;
  readonly taskInstance?: TaskInstanceResponse;
  readonly withText?: boolean;
};

const ClearTaskInstanceButton = ({
  groupTaskInstance,
  isHotkeyEnabled = false,
  onOpen,
  taskInstance,
  withText = true,
}: Props) => {
  const { onClose, onOpen: onOpenInternal, open } = useDisclosure();
  const { t: translate } = useTranslation();
  const isGroup = groupTaskInstance && !taskInstance;
  const useInternalDialog = !Boolean(onOpen);

  const selectedInstance = taskInstance ?? groupTaskInstance;

  useHotkeys(
    "shift+c",
    () => {
      if (onOpen && selectedInstance) {
        onOpen(selectedInstance);
      } else {
        onOpenInternal();
      }
    },
    { enabled: isHotkeyEnabled },
  );

  return (
    <Tooltip
      closeDelay={100}
      content={translate("dags:runAndTaskActions.clear.buttonTooltip")}
      disabled={!isHotkeyEnabled}
      openDelay={100}
    >
      <Box>
        <ActionButton
          actionName={translate("dags:runAndTaskActions.clear.button", {
            type: translate("taskInstance_one"),
          })}
          icon={<CgRedo />}
          onClick={() => (onOpen && selectedInstance ? onOpen(selectedInstance) : onOpenInternal())}
          text={translate("dags:runAndTaskActions.clear.button", {
            type: translate(isGroup ? "taskGroup" : "taskInstance_one"),
          })}
          withText={withText}
        />

        {useInternalDialog && open && isGroup ? (
          <ClearGroupTaskInstanceDialog onClose={onClose} open={open} taskInstance={groupTaskInstance} />
        ) : undefined}

        {useInternalDialog && open && !isGroup && taskInstance ? (
          <ClearTaskInstanceDialog onClose={onClose} open={open} taskInstance={taskInstance} />
        ) : undefined}
      </Box>
    </Tooltip>
  );
};

export default ClearTaskInstanceButton;
